#!/usr/bin/env python
# -*- coding:utf-8 -*-
from .embeddings import *


class LLMFactory:
    EMBEDDING_MAPPING = {
        'zhipu': ZhiPuEmbedding,
    }

    @classmethod
    def get_embedding(cls, model_name: str, api_key: str):
        return cls.EMBEDDING_MAPPING[model_name](api_key).embedding
